import numpy as np
import cv2
import matplotlib.pyplot as plt
def FFT (path):
    img=plt.imread (path)
    plt.subplot(121)
    plt. imshow(img)
    plt.title('img')
    #转化为灰度图
    img = 0.2126 * img[:,:,0] + 0.7152*img[:,:,1] + 0.0722 * img[:,:,2]
    #快速傅里叶变换
    img_fft =np.fft.fft2(img)
    #原点移动
    img_fft = np.fft.fftshift(img_fft)
    #对数变换
    img_fft = np. log(1 + np.abs(img_fft))
    plt.subplot(122)
    plt.imshow(img_fft, 'gray')
    plt.title('img_fft')
    plt.show()
#FFT("lena. jpg')
#FFT('十字图．jpg'
FFT('./img/1.jpg')